pandas 和 numpy 中 where 使用 |
您所在的位置：网站首页 › dataframe 筛选 in › pandas 和 numpy 中 where 使用 |
|
注意： df1.where(cond,df2) 等价于 np.where(cond, df1, df2)
1. pandas.DataFrame.where
DataFrame.where(cond, other=nan, inplace=False, axis=None, level=None, try_cast=False, raise_on_error=True) 功能：按条件查找替换，cond 为 True 则 self 值保持不变；False 改为参数 other 对应的值。结果返回与 self 相同的对象。 参数： cond 查找条件 other cond为False时要替换的值 inplace 是否在原数据上操作 >>> import numpy as np >>> import pandas as pd >>> data = pd.DataFrame(np.arange(12).reshape(3,4),columns=['a','b','c','d']) >>> data a b c d 0 0 1 2 3 1 4 5 6 7 2 8 9 10 11 >>> data.where(data['a']>3, 3.0) a b c d 0 3 3 3 3 1 4 5 6 7 2 8 9 10 11 2.numpy.where声明： 首先强调一下，where()函数对于不同的输入，返回值是不同的。 当数组是一维数组时，返回的值是一维的索引，所以只有一组索引数组 当数组是多维数组时，满足条件的数组值返回的是值的位置索引，因此会有两组索引数组来表示值的位置。 >>> import numpy as np >>> a = np.arange(10) >>> a array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) >>> np.where(a>5) (array([6, 7, 8, 9], dtype=int64),) >>> b = np.arange(20).reshape(4,5) >>> b array([[ 0, 1, 2, 3, 4], [ 5, 6, 7, 8, 9], [10, 11, 12, 13, 14], [15, 16, 17, 18, 19]]) >>> np.where(b>10) (array([2, 2, 2, 2, 3, 3, 3, 3, 3], dtype=int64), array([1, 2, 3, 4, 0, 1, 2, 3, 4], dtype=int64)) 具体实现numpy.where(condition[, x, y]) 功能： 参数： condition： 判定条件，如果True，选择 x；False，选择y（数据类型为数组，bool 值）x,y（可选）： x 和 y 的 shape 必须和 condition 相同（可以采用 broadcast，广播机制） ①如果参数有condition，x和y，它们三个参数的shape是相同的。那么，当condition中的值是true时返回x对应位置的值，false是返回y的。 ②如果参数只有condition的话，返回值是condition中元素值为true的位置索引，且是以元组形式返回，元组的元素是ndarray数组，表示位置的索引 >>> np.where([[True, False], [True, True]], # 官网例子 ... [[1, 2], [3, 4]], ... [[9, 8], [7, 6]]) array([[1, 8], [3, 4]]) >>> x = np.arange(9).reshape(3, 3) >>> np.where(x>5) # ndarray 数组分别表示对应的 行和列 (array([2, 2, 2], dtype=int64), array([0, 1, 2], dtype=int64))https://www.cnblogs.com/zz22–/p/8654395.html |
今日新闻 |
推荐新闻 |
| CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |